package com.example.mq.mqserver.core;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;
import lombok.ToString;
import lombok.extern.slf4j.Slf4j;

import java.util.HashMap;
import java.util.Map;

/*
 * 这个类表示一个交换机
 */
@Data
@Slf4j
@ToString
public class Exchange {
    // 此处使用 name 来作为交换机的身份标识 (唯一)
    private String name;

    // 交换机类型, DIRECT, FANOUT, TOPIC
    private ExchangeType type = ExchangeType.DIRECT;

    // 该交换机是否要持久化存储. true 表示需要持久化; false 表示不必持久化
    private boolean durable = false;

    // 如果当前交换机没人使用了, 就会自动被删除 (RabbitMQ 实现了这个功能, 这里未实现)
    private boolean autoDelete = false;

    // 创建交换机时指定的一些额外的参数选项 arguments (RabbitMQ 实现了这个功能, 这里未实现)
    // 为了把这个 arguments 存到数据库中, 需要把 Map 转成 JSON 格式的字符串
    private Map<String, Object> arguments = new HashMap<>();

    public Exchange(String name) {
        this.name = name;
    }

    // 从 Map 转成 JSON 格式的字符串，用于和数据库交互
    public String getArguments() {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            log.error(e.getMessage());
            // 异常时返回一个空的 JSON 字符串
            return "{}";
        }
    }

    // 从数据库读取数据后，构造 Exchange 对象时会自动调用
    // 将 JSON 字符串解析成 Map 对象
    public void setArguments(String argumentsJson) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String, Object>>() {
            });
        } catch (JsonProcessingException e) {
            log.error(e.getMessage());
        }
    }

    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }

    // 针对 arguments 提供一组 getter 和 setter，用来方便地获取/设置键值对 (内部使用)
    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArgumentsByKV(String key, Object value) {
        arguments.put(key, value);
    }
}
